package com.example.springboot.controller;

import cn.hutool.core.util.StrUtil;
import cn.hutool.poi.excel.ExcelReader;
import cn.hutool.poi.excel.ExcelUtil;
import cn.hutool.poi.excel.ExcelWriter;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.example.springboot.common.HoneyLogs;
import com.example.springboot.common.LogType;
import com.example.springboot.common.Result;
import com.example.springboot.exception.ServiceException;
import com.example.springboot.mapper.UserMapper;
import com.example.springboot.service.UserService;
import com.example.springboot.utils.TokenUtils;
import org.apache.ibatis.annotations.Update;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.dao.DuplicateKeyException;
import org.springframework.web.bind.annotation.*;
import com.example.springboot.entity.User;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import javax.websocket.server.PathParam;
import java.io.IOException;
import java.net.URLEncoder;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.Optional;
import java.util.stream.Collectors;

/**
 * 功能：
 * 作者帆哥
 * 日期： 2024/8/9 22:55
 */
//@CrossOrigin
//@RestController
//@RequestMapping("/user")
//public class UserController {
//
//    @Autowired
//    UserService userService;
//
//    /**
//     * 新增用户信息
//     */
//    @PostMapping("/add")
//    public Result add(@RequestBody User user) {
//        try {
//            userService.insertUser(user);
//        } catch (Exception e) {
//            if (e instanceof DuplicateKeyException) {
//                return Result.error("插入数据库失败");
//            } else {
//                return Result.error("系统错误");
//            }
//
//        }
//
//        return Result.success();
//    }
//
//    /**
//     * 修改用户信息
//     */
//    @PutMapping("/update")
//    public Result update(@RequestBody User user) {
//        userService.updateUser(user);
//        return Result.success();
//    }
//
//
//    /**
//     * 单个删除用户信息
//     */
//    @DeleteMapping("/delete/{id}")
//    public Result delete(@PathVariable Integer id) {
//        userService.deleteUser(id);
//        return Result.success();
//    }
//
//
//    /**
//     * 批量删除用户信息
//     */
//    @DeleteMapping("/delete/batch")
//    public Result batchDelete(@RequestBody List<Integer> ids) {
//        userService.batchDeleteUser(ids);
//        return Result.success();
//    }
//
//
//    /**
//     * 查询全部用户信息
//     */
//    @GetMapping("/selectAll")
//    public Result selectAll() {
//        List<User> userList = userService.selectAll();
//        return Result.success(userList);
//    }
//
//    /**
//     * 根据单个id条件查询(具有唯一性)
//     */
//    @GetMapping("/selectById/{id}")
//    public Result selectById(@PathVariable Integer id) {
//        User user = userService.selectById(id);
//        return Result.success(user);
//    }
//
//    /**
//     * 根据单个name条件查询（不惧有唯一性，具备多个）
//     * 根据条件查询的时候，如果不确定查询的结果,那就统一返回一个List对象集合
//     * 这样是最稳妥的方式
//     * 不会出现错误
//     */
//    @GetMapping("/selectByName/{name}")
//    public Result selectByName(@PathVariable String name) {
//        List<User> userList = userService.selectByName(name);
//        return Result.success(userList);
//    }
//
//    /**
//     * 多条件查询
//     */
//    @GetMapping("/selectByMore")
//    public Result selectByMore(@RequestParam String username, @RequestParam String name) {
//        List<User> userList = userService.selectByMore(username, name);
//        return Result.success(userList);
//    }
//
//
//    /**
//     * 多条件模糊查询
//     */
//    @GetMapping("/selectByMo")
//    public Result selectByMo(@RequestParam String username, @RequestParam String name) {
//        List<User> userList = userService.selectByMo(username, name);
//        return Result.success(userList);
//    }
//
//    /**
//     * 多条件分页模糊查询
//     */
//    @GetMapping("/selectByPage")
//    public Result selectByPage(
//            @RequestParam Integer pageNum,
//            @RequestParam Integer pageSize,
//            @RequestParam String username,
//            @RequestParam String name) {
//        Page<User> userList = userService.selectByPage(pageNum, pageSize, username, name);
//        return Result.success(userList);
//    }
//
//}

/**
 * mybastit 改造
 */
@CrossOrigin
@RestController
@RequestMapping("/user")
public class UserController {

    @Autowired
    UserService userService;

    /**
     * 新增用户信息
     */
    @HoneyLogs(operation = "用户", type= LogType.ADD)
    @PostMapping("/add")
    public Result add(@RequestBody User user) {
        try {
            userService.save(user);
        } catch (Exception e) {
            if (e instanceof DuplicateKeyException) {
                return Result.error("插入数据库失败");
            } else {
                return Result.error("系统错误");
            }

        }

        return Result.success();
    }

    /**
     * 修改用户信息
     */
    @HoneyLogs(operation = "用户", type= LogType.UPDATE)
    @PutMapping("/update")
    public Result update(@RequestBody User user) {
        userService.updateById(user);
        return Result.success();
    }


    /**
     * 单个删除用户信息
     */
    @HoneyLogs(operation = "用户", type= LogType.DELETE)
    @DeleteMapping("/delete/{id}")
    public Result delete(@PathVariable Integer id) {
        User currentUser = TokenUtils.getCurrentUser();
        if (id.equals(currentUser.getId())) {
            throw new ServiceException("不能删除当前的用户");
        }

        //以上是用于防止用户删除当前的管理员登录的账号
        userService.removeById(id);
        return Result.success();
    }


    /**
     * 批量删除用户信息
     */
    @HoneyLogs(operation = "用户", type= LogType.BATCH_DELETE)
    @DeleteMapping("/delete/batch")
    public Result batchDelete(@RequestBody List<Integer> ids) {
        User currentUser = TokenUtils.getCurrentUser();
        if (currentUser != null && currentUser.getId() != null && ids.contains(currentUser.getId())) {
            throw new ServiceException("不能删除当前的用户");
        }

        //以上是用于防止用户删除当前的管理员登录的账号
        userService.removeBatchByIds(ids);
        return Result.success();
    }


    /**
     * 查询全部用户信息
     */
    @GetMapping("/selectAll")
    public Result selectAll() {
        //select * from user order by id desc    (list方法获取所有)
        List<User> userList = userService.list(new QueryWrapper<User>().orderByDesc("id"));
        return Result.success(userList);
    }

    /**
     * 根据单个id条件查询(具有唯一性)
     */
    @GetMapping("/selectById/{id}")
    public Result selectById(@PathVariable Integer id) {
        User user = userService.getById(id);
        return Result.success(user);
    }

    /**
     * 根据单个name条件查询（不惧有唯一性，具备多个）
     * 根据条件查询的时候，如果不确定查询的结果,那就统一返回一个List对象集合
     * 这样是最稳妥的方式
     * 不会出现错误         可能用不到
     */
//    @GetMapping("/selectByName/{name}")
//    public Result selectByName(@PathVariable String name) {
//        List<User> userList = userService.selectByName(name);
//        return Result.success(userList);
//    }

    /**
     * 多条件查询
     */
//    @GetMapping("/selectByMore")
//    public Result selectByMore(@RequestParam String username, @RequestParam String name) {
//        List<User> userList = userService.selectByMore(username, name);
//        return Result.success(userList);
//    }


    /**
     * 多条件模糊查询
     */
//    @GetMapping("/selectByMo")
//    public Result selectByMo(@RequestParam String username, @RequestParam String name) {
//        List<User> userList = userService.selectByMo(username, name);
//        return Result.success(userList);
//    }

    /**
     * 多条件分页模糊查询
     */
    @GetMapping("/selectByPage")
    public Result selectByPage(
            @RequestParam Integer pageNum,
            @RequestParam Integer pageSize,
            @RequestParam String username,
            @RequestParam String name) {
        QueryWrapper<User> queryWrapper = new QueryWrapper<User>().orderByDesc("id");
        queryWrapper.like(StrUtil.isNotBlank(username), "username", username);
        queryWrapper.like(StrUtil.isNotBlank(name), "name", name);
        //IPage<User> page = new Page<User>(pageNum, pageSize);
        Page<User> userList = userService.page(new Page<User>(pageNum, pageSize), queryWrapper);
        return Result.success(userList);
    }


    /**
     * 批量导出
     */
    @GetMapping("/export")
    public void exportData(
            @RequestParam(required = false) String username,
            @RequestParam(required = false) String name,
            @RequestParam(required = false) String ids,
            HttpServletResponse response) throws IOException {
        ExcelWriter writer = ExcelUtil.getWriter(true);
        List<User> list;
        QueryWrapper<User> queryWrapper = new QueryWrapper<>();
        if (StrUtil.isNotBlank(ids)) {   //["1","2","3"]   ---> [1,2,3]
            List<Integer> idsArr1 = Arrays.stream(ids.split(",")).map(Integer::valueOf).collect(Collectors.toList());
            queryWrapper.in("id", idsArr1);
        } else {
            //第一种全部导出  全部导出和搜索条件的导出
            queryWrapper.like(StrUtil.isNotBlank(username), "username", username);
            queryWrapper.like(StrUtil.isNotBlank(name), "name", name);
        }

        list = userService.list(queryWrapper); //查询当前User表的所有数据
        writer.write(list, true);

        response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=utf-8");
        response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode("用户信息表", "UTF-8") + ".xlsx");
        ServletOutputStream outputStream = response.getOutputStream();
        writer.flush(outputStream, true);
        writer.close();
        outputStream.flush();
        outputStream.close();

    }


    /***
     *
     * 批量导入
     * @param file 文件对象
     * @return 导出结果
     */
    @PostMapping("/import")
    public Result importData(MultipartFile file) throws IOException {
        ExcelReader reader = ExcelUtil.getReader(file.getInputStream());
        List<User> userList = reader.readAll(User.class);
        //写入数据
        try {
            userService.saveBatch(userList);
        } catch (Exception e) {
            e.printStackTrace();
            return Result.error("批量导入错误");
        }
        return Result.success();
    }
}
